2d image to 3d in python